package InternetCode.Socket.Exa9;

import java.io.*;
import java.net.*;
import java.nio.charset.StandardCharsets;
import java.util.Scanner;

/**
 * 解决Socket传输数据过程中的乱码问题
 */
public class ClientSocketFrame {

    //输出流
    private OutputStreamWriter out;
    //输出流
    private PrintWriter writer;
    //输入流
    private BufferedReader reader;
    //套接字
    private Socket socket;
    private Scanner scanner;


    /**
     * 和服务器创建连接
     */
    private void connect() {
        System.out.println("尝试连接......\n");
        try {
            // 实例化Socket对象,创建连接
            socket = new Socket("127.0.0.1", 1978);
            while (true) {
                //从连接中获得输出流并包装进Writer中，这里通过设置字符集来避免乱码
                out = new OutputStreamWriter(socket.getOutputStream(), StandardCharsets.UTF_8);
                writer = new PrintWriter(out, true);
                // 实例化缓冲输入流对象，这里通过设置字符集来避免乱码
                reader = new BufferedReader(new InputStreamReader(socket.getInputStream(), StandardCharsets.UTF_8));
                System.out.println("完成连接。\n");
                getServerInfo();
            }
        } catch (Exception e) {
            // 输出异常信息
            e.printStackTrace();
        }
    }

    /**
     * 监听服务器发送来的消息
     */
    private void getServerInfo() {
        try {
            while (true) {
                if (reader != null) {
                    String line = reader.readLine();
                    if (line != null)
                        System.out.println("接收到服务器发送的信息：" + line + "\n"); // 获得客户端信息
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (reader != null) {
                    reader.close();// 关闭流
                }
                if (socket != null) {
                    socket.close(); // 关闭套接字
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 向服务器发送消息方法
     */
    private void sendInfoToServer(){
        while(true){
            scanner=new Scanner(System.in);
            System.out.println("请输入要发送的消息:");
            String text=scanner.nextLine();
            if(text.equals("quit")){
                break;
            }
            writer.println(text);
        }
    }

    public static void main(String[] args) { // 主方法
        ClientSocketFrame client = new ClientSocketFrame(); // 创建本例对象
        //此线程用于监听接收消息
        new Thread(new Runnable() {
            @Override
            public void run() {
                client.connect(); // 调用连接方法
            }
        }).start();
        //此线程用发送消息给服务器
        new Thread(new Runnable() {
            @Override
            public void run() {
                client.sendInfoToServer(); // 调用连接方法
            }
        }).start();
    }

}
